from django import template


register = template.Library()


@register.simple_tag
def get_liked_images(user):

    """ Получаем понравившиеся изображения """

    liked_images = [like.content_object for like in user.liked.all()]
    return liked_images


@register.simple_tag
def is_like(user, image):

    """ Проверяем был ли поставлен лайк изображению """

    liked_image = get_liked_images(user)

    if image in liked_image: return True
    else: return False


@register.simple_tag
def get_number_of_like(image):

    """ Получаем количество лайков фотографии """

    number_of_likes = image.likes.count()
    return number_of_likes



